#!/usr/bin/env python
# -*- coding: UTF-8 -*-
# @Project : RAG-demo
# @File    : schema.py
# @IDE     : PyCharm
# @Author  :ZH
# @Time    : 2025/6/27 16:44
from typing import Optional

from pydantic import Field, BaseModel


class MessageSchema(BaseModel):
    """
    创建聊天模型请求
    """
    ai_message: str = Field(description='AI消息')
    human_message: str = Field(description='用户消息')
    token_usage: dict = Field(default={}, description='token消耗')




